Module 12: Maps

Let's draw some maps. 🗺🧐

A dotmap with Altair

Let's start with altair. When your dataset is large, it is nice to enable a json data transformer. What it does is, instead of generating and holding the whole dataset in the memory, transform the dataset and save into a temporary file. This makes the whole plotting process much more efficient. For more information, check out: https://altair-viz.github.io/user_guide/data_transformers.html

In [1]:
import altair as alt

# saving data into a file rather than embedding into the chart
alt.data_transformers.enable('json')

#alt.renderers.enable('notebook')
# alt.renderers.enable('jupyterlab')
alt.renderers.enable('default')
Out[1]:
RendererRegistry.enable('default')

Maybe we need a dataset with geographical coordinates. This zipcodes dataset contains the location and zipcode of each zip code area.

In [2]:
from vega_datasets import data

zipcodes_url = data.zipcodes.url
zipcodes = data.zipcodes()
zipcodes.head()
Out[2]:
zip_code latitude longitude city state county
0 00501 40.922326 -72.637078 Holtsville NY Suffolk
1 00544 40.922326 -72.637078 Holtsville NY Suffolk
2 00601 18.165273 -66.722583 Adjuntas PR Adjuntas
3 00602 18.393103 -67.180953 Aguada PR Aguada
4 00603 18.455913 -67.145780 Aguadilla PR Aguadilla
In [3]:
zipcodes.shape
Out[3]:
(42049, 6)
In [4]:
zipcodes = data.zipcodes(dtype={'zip_code': 'category'})
zipcodes.head()
Out[4]:
zip_code latitude longitude city state county
0 00501 40.922326 -72.637078 Holtsville NY Suffolk
1 00544 40.922326 -72.637078 Holtsville NY Suffolk
2 00601 18.165273 -66.722583 Adjuntas PR Adjuntas
3 00602 18.393103 -67.180953 Aguada PR Aguada
4 00603 18.455913 -67.145780 Aguadilla PR Aguadilla
In [5]:
zipcodes.dtypes
Out[5]:
zip_code     category
latitude      float64
longitude     float64
city           object
state          object
county         object
dtype: object
In [6]:
zipcodes.zip_code.dtype
Out[6]:
CategoricalDtype(categories=['00501', '00544', '00601', '00602', '00603', '00604',
                  '00605', '00606', '00610', '00611',
                  ...
                  '99919', '99921', '99922', '99923', '99925', '99926',
                  '99927', '99928', '99929', '99950'],
, ordered=False)
In [7]:
zipcodes.county.dtype
Out[7]:
dtype('O')

Btw, you'll have fewer issues if you pass URL instead of a dataframe to alt.Chart.

Let's draw it

Now we have the dataset loaded and start drawing some plots. Let's say you don't know anything about map projections. What would you try with geographical data? Probably the simplest way is considering (longitude, latitude) as a Cartesian coordinate and directly plot them.

In [8]:
alt.Chart(zipcodes_url).mark_circle().encode(
    x='longitude:Q',
    y='latitude:Q',
)
Out[8]:

Actually this itself is a map projection called Equirectangular projection. This projection (or almost a non-projection) is super straight-forward and doesn't require any processing of the data. So, often it is used to just quickly explore geographical data. As you dig deeper, you still want to think about which map projection fits your need best. Don't just use equirectangular projection without any thoughts!

Anyway, let's make it look slighly better by reducing the size of the circles and adjusting the aspect ratio.

Q: Can you adjust the circle size, width and height of the chart?

In [9]:
alt.Chart(zipcodes_url).mark_circle(size=3).encode(
    x='longitude:Q',
    y='latitude:Q',
).properties(
    width=700,
    height=200
)
Out[9]:

But, a much better way to do this is explicitly specifying that they are lat, lng coordinates by using longitude= and latitude=, rather than x= and y=. If you do that, altair automatically adjust the aspect ratio.

Q: Can you try it?

In [10]:
alt.Chart(zipcodes_url).mark_circle(size=3).encode(
    longitude='longitude:Q',
    latitude='latitude:Q',
).properties(
    width=800,
    height=300
)
Out[10]:

Because the American empire is far-reaching and complicated, the information density of this map is very low (although interesting). A common projection for visualizing US data is AlbersUSA, which uses Albers (equal-area) projection. This is a standard projection used in United States Geological Survey and the United States Census Bureau. Albers USA contains a composition of US main land, Alaska, and Hawaii.

To use it, we call project method and specify which variables are longitude and latitude.

Q: use the project method to draw the map in the AlbersUsa projection.

In [11]:
alt.Chart(zipcodes_url).mark_circle(size=3).encode(
    longitude='longitude:Q',
    latitude='latitude:Q',
     tooltip='zip_code:N'
).project(
    type='albersUsa'
).properties(
    width=700,
    height=400
)
Out[11]:

Now we're talking. 😎

Let's visualize the large-scale zipcode patterns. We can use the fact that the zipcodes are hierarchically organized. That is, the first digit captures the largest area divisions and the other digits are about smaller geographical divisions.

Altair provides some data transformation functionalities. One of them is extracting a substring from a variable.

In [12]:
from altair.expr import datum, substring

alt.Chart(zipcodes_url).mark_circle(size=2).transform_calculate(
    'first_digit', substring(datum.zip_code, 0, 1)
).encode(
    longitude='longitude:Q',
    latitude='latitude:Q',
    color='first_digit:N',
).project(
    type='albersUsa'
).properties(
    width=700,
    height=400,
)
Out[12]:

For each row (datum), you obtain the zip_code variable and get the substring (imagine Python list slicing), and then you call the result first_digit. Now, you can use this first_digit variable to color the circles. Also note that we specify first_digit as a nominal variable, not quantitative, to obtain a categorical colormap. But we can also play with it too.

Q: Why don't you extract the first two digits, name it as two_digits, and declare that as a quantitative variable? Any interesting patterns? What does it tell us about the history of US?

In [13]:
from altair.expr import datum, substring

alt.Chart(zipcodes_url).mark_circle(size=2).transform_calculate(
    'two_digit', substring(datum.zip_code,0,2)
).encode(
    longitude='longitude:Q',
    latitude='latitude:Q',
    color='two_digit:Q',
).project(
    type='albersUsa'
).properties(
    width=700,
    height=400,
)
Out[13]:

Q: also try it with declaring the first two digits as a categorical variable

In [14]:
from altair.expr import datum, substring

alt.Chart(zipcodes_url).mark_circle(size=2).transform_calculate(
    'two_digit', substring(datum.zip_code,0,2)
).encode(
    longitude='longitude:Q',
    latitude='latitude:Q',
    color='two_digit:N',
).project(
    type='albersUsa'
).properties(
    width=700,
    height=400,
)
Out[14]:

Btw, you can always click "view source" or "open in Vega Editor" to look at the json object that defines this visualization. You can embed this json object on your webpage and easily put up an interactive visualization.

Q: Can you put a tooltip that displays the zipcode when you mouse-over? Example https://altair-viz.github.io/gallery/scatter_tooltips.html

In [15]:
from altair.expr import datum, substring

alt.Chart(zipcodes_url).mark_circle(size=2).transform_calculate(
    'first_digit', substring(datum.zip_code,0,1)
).encode(
    longitude='longitude:Q',
    latitude='latitude:Q',
    color='first_digit:N',
    tooltip='zip_code:N'
).project(
    type='albersUsa'
).properties(
    width=700,
    height=400,
)
Out[15]:
In [16]:
from altair.expr import datum, substring

alt.Chart(zipcodes_url).mark_circle(size=2).transform_calculate(
    'two_digit', substring(datum.zip_code,0,2)
).encode(
    longitude='longitude:Q',
    latitude='latitude:Q',
    color='two_digit:N',
    tooltip='zip_code:N'
).project(
    type='albersUsa'
).properties(
    width=700,
    height=400,
)
Out[16]:

Choropleth

Let's try some choropleth now. Vega datasets have US county / state boundary data (us_10m) and world country boundary data (world-110m). You can take a look at the boundaries on GitHub (they renders topoJSON files):

If you click "Raw" then you can take a look at the actual file, which is hard to read.

Essentially, each file is a large dictionary with the following keys.

In [17]:
usmap = data.us_10m()
usmap.keys()
Out[17]:
dict_keys(['type', 'transform', 'objects', 'arcs'])
In [18]:
usmap['type']
Out[18]:
'Topology'
In [19]:
usmap['transform']
Out[19]:
{'scale': [0.003589294092944858, 0.0005371535195261037],
 'translate': [-179.1473400003406, 17.67439566600018]}

This transformation is used to quantize the data and store the coordinates in integer (easier to store than float type numbers).

https://github.com/topojson/topojson-specification#212-transforms

In [20]:
usmap['objects'].keys()
Out[20]:
dict_keys(['counties', 'states', 'land'])

This data contains not only county-level boundaries (objects) but also states and land boundaries.

In [21]:
usmap['objects']['land']['type'], usmap['objects']['states']['type'], usmap['objects']['counties']['type']
Out[21]:
('MultiPolygon', 'GeometryCollection', 'GeometryCollection')

land is a multipolygon (one object) and states and counties contains many geometrics (multipolygons) because there are many states (counties). We can look at a state as a set of arcs that define it. It's id captures the identity of the state and is the key to link to other datasets.

In [22]:
state1 = usmap['objects']['states']['geometries'][1]
state1
Out[22]:
{'arcs': [[[10337]],
  [[10342]],
  [[10341]],
  [[10343]],
  [[10834, 10340]],
  [[10344]],
  [[10345]],
  [[10338]]],
 'id': 15,
 'type': 'MultiPolygon'}

The arcs referred here is defined in usmap['arcs'].

In [23]:
usmap['arcs'][:10]
Out[23]:
[[[15739, 57220], [0, 0]],
 [[15739, 57220], [29, 62], [47, -273]],
 [[15815, 57009], [-6, -86]],
 [[15809, 56923], [0, 0]],
 [[15809, 56923], [-36, -8], [6, -210], [32, 178]],
 [[15811, 56883], [9, -194], [44, -176], [-29, -151], [-24, -319]],
 [[15811, 56043], [-12, -216], [26, -171]],
 [[15825, 55656], [-2, 1]],
 [[15823, 55657], [-19, 10], [26, -424], [-26, -52]],
 [[15804, 55191], [-30, -72], [-47, -344]]]

It seems pretty daunting to work with this dataset, right? But fortunately people have already built tools to handle such data.

In [24]:
# states
states = alt.topo_feature(data.us_10m.url, 'states')

# us counties
us_counties = alt.topo_feature(data.us_10m.url, 'counties')
In [25]:
states
Out[25]:
UrlData({
  format: TopoDataFormat({
    feature: 'states',
    type: 'topojson'
  }),
  url: 'https://cdn.jsdelivr.net/npm/vega-datasets@v1.29.0/data/us-10m.json'
})

Q. Can you find a mark for geographical shapes from here https://altair-viz.github.io/user_guide/marks.html and draw the states?

In [26]:
alt.Chart(states).mark_geoshape(
        stroke='white',
        strokeWidth=0.1).properties(
    width=500,
    height=300
)
Out[26]:

And then project it using the albersUsa?

In [27]:
alt.Chart(states).mark_geoshape(
        stroke='white',
        strokeWidth=0.1).project(
    type='albersUsa'
).properties(
    width=500,
    height=300
)
Out[27]:

Can you do the same thing with counties and draw county boundaries? (hint: you have to use alt.topo_feature())

In [28]:
states = alt.Chart(states).mark_geoshape(
        stroke='white',
        strokeWidth=0.1).project(
    type='albersUsa'
).properties(
    width=500,
    height=300
)

counties = alt.Chart(us_counties).mark_geoshape(
        stroke='white',
        strokeWidth=0.1
    ).project('albersUsa')

states + counties
Out[28]:

Let's load some county-level unemployment data.

In [29]:
unemp_data = data.unemployment(sep='\t')
unemp_data.head()
Out[29]:
id rate
0 1001 0.097
1 1003 0.091
2 1005 0.134
3 1007 0.121
4 1009 0.099

This dataset has unemployment rate. When? I don't know. We don't care about data provenance here because the goal is quickly trying out choropleth. But if you're working with a real dataset, you should be very sensitive about the provenance of your dataset. Make sure you understand where the data came from and how it was processed.

Anyway, for each county specified with id. To combine two datasets, we use "Lookup transform" - https://vega.github.io/vega/docs/transforms/lookup/. Essentially, we use the id in the map data to look up (again) id field in the unemp_data and then bring in the rate variable. Then, we can use that rate variable to encode the color of the geoshape mark.

In [30]:
alt.Chart(us_counties).mark_geoshape().project(
    type='albersUsa'
).transform_lookup(
    lookup='id',
    from_=alt.LookupData(data.unemployment.url, 'id', ['rate'])
).encode(
    color='rate:Q'
).properties(
    width=700,
    height=400
)
Out[30]:

There you have it, a nice choropleth map. 😎

Raster visualization with datashader

Although many geovisualizations use vector graphics, raster visualization is still useful especially when you deal with images and lots of datapoints. Datashader is a package that aggregates and visualizes a large amount of data very quickly. Given a scene (visualization boundary, resolution, etc.), it quickly aggregate the data and produce pixels and send them to you.

To appreciate its power, we need a fairly large dataset. Let's use NYC taxi trip dataset on Kaggle: https://www.kaggle.com/kentonnlp/2014-new-york-city-taxi-trips You can download even bigger trip data from NYC open data website: https://opendata.cityofnewyork.us/data/

Ah, and you want to install the datashader, bokeh, and holoviews first if you don't have them yet. If you have them make sure they are the latest version

pip install -U datashader bokeh holoviews

or

conda install datashader bokeh holoviews
In [31]:
!pip install -U datashader bokeh holoviews
Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/
Requirement already satisfied: datashader in /usr/local/lib/python3.7/dist-packages (0.14.1)
Requirement already satisfied: bokeh in /usr/local/lib/python3.7/dist-packages (2.4.3)
Requirement already satisfied: holoviews in /usr/local/lib/python3.7/dist-packages (1.15.0)
Requirement already satisfied: numpy>=1.7 in /usr/local/lib/python3.7/dist-packages (from datashader) (1.21.6)
Requirement already satisfied: scipy in /usr/local/lib/python3.7/dist-packages (from datashader) (1.4.1)
Requirement already satisfied: pandas>=0.24.1 in /usr/local/lib/python3.7/dist-packages (from datashader) (1.3.5)
Requirement already satisfied: datashape>=0.5.1 in /usr/local/lib/python3.7/dist-packages (from datashader) (0.5.2)
Requirement already satisfied: xarray>=0.9.6 in /usr/local/lib/python3.7/dist-packages (from datashader) (0.20.2)
Requirement already satisfied: dask[complete]>=0.18.0 in /usr/local/lib/python3.7/dist-packages (from datashader) (2.12.0)
Requirement already satisfied: pillow>=3.1.1 in /usr/local/lib/python3.7/dist-packages (from datashader) (7.1.2)
Requirement already satisfied: pyct>=0.4.5 in /usr/local/lib/python3.7/dist-packages (from datashader) (0.4.8)
Requirement already satisfied: param>=1.6.1 in /usr/local/lib/python3.7/dist-packages (from datashader) (1.12.1)
Requirement already satisfied: numba>=0.51 in /usr/local/lib/python3.7/dist-packages (from datashader) (0.51.2)
Requirement already satisfied: colorcet>=0.9.0 in /usr/local/lib/python3.7/dist-packages (from datashader) (3.0.0)
Requirement already satisfied: cloudpickle>=0.2.1 in /usr/local/lib/python3.7/dist-packages (from dask[complete]>=0.18.0->datashader) (2.1.0)
Requirement already satisfied: PyYaml in /usr/local/lib/python3.7/dist-packages (from dask[complete]>=0.18.0->datashader) (3.13)
Requirement already satisfied: fsspec>=0.6.0 in /usr/local/lib/python3.7/dist-packages (from dask[complete]>=0.18.0->datashader) (2022.5.0)
Requirement already satisfied: partd>=0.3.10 in /usr/local/lib/python3.7/dist-packages (from dask[complete]>=0.18.0->datashader) (1.2.0)
Requirement already satisfied: toolz>=0.7.3 in /usr/local/lib/python3.7/dist-packages (from dask[complete]>=0.18.0->datashader) (0.11.2)
Requirement already satisfied: distributed>=2.0 in /usr/local/lib/python3.7/dist-packages (from dask[complete]>=0.18.0->datashader) (2.30.1)
Requirement already satisfied: typing-extensions>=3.10.0 in /usr/local/lib/python3.7/dist-packages (from bokeh) (4.1.1)
Requirement already satisfied: Jinja2>=2.9 in /usr/local/lib/python3.7/dist-packages (from bokeh) (2.11.3)
Requirement already satisfied: packaging>=16.8 in /usr/local/lib/python3.7/dist-packages (from bokeh) (21.3)
Requirement already satisfied: tornado>=5.1 in /usr/local/lib/python3.7/dist-packages (from bokeh) (5.1.1)
Requirement already satisfied: multipledispatch>=0.4.7 in /usr/local/lib/python3.7/dist-packages (from datashape>=0.5.1->datashader) (0.6.0)
Requirement already satisfied: python-dateutil in /usr/local/lib/python3.7/dist-packages (from datashape>=0.5.1->datashader) (2.8.2)
Requirement already satisfied: sortedcontainers!=2.0.0,!=2.0.1 in /usr/local/lib/python3.7/dist-packages (from distributed>=2.0->dask[complete]>=0.18.0->datashader) (2.4.0)
Requirement already satisfied: tblib>=1.6.0 in /usr/local/lib/python3.7/dist-packages (from distributed>=2.0->dask[complete]>=0.18.0->datashader) (1.7.0)
Requirement already satisfied: click>=6.6 in /usr/local/lib/python3.7/dist-packages (from distributed>=2.0->dask[complete]>=0.18.0->datashader) (7.1.2)
Requirement already satisfied: setuptools in /usr/local/lib/python3.7/dist-packages (from distributed>=2.0->dask[complete]>=0.18.0->datashader) (57.4.0)
Requirement already satisfied: psutil>=5.0 in /usr/local/lib/python3.7/dist-packages (from distributed>=2.0->dask[complete]>=0.18.0->datashader) (5.4.8)
Requirement already satisfied: zict>=0.1.3 in /usr/local/lib/python3.7/dist-packages (from distributed>=2.0->dask[complete]>=0.18.0->datashader) (2.2.0)
Requirement already satisfied: msgpack>=0.6.0 in /usr/local/lib/python3.7/dist-packages (from distributed>=2.0->dask[complete]>=0.18.0->datashader) (1.0.4)
Requirement already satisfied: MarkupSafe>=0.23 in /usr/local/lib/python3.7/dist-packages (from Jinja2>=2.9->bokeh) (2.0.1)
Requirement already satisfied: six in /usr/local/lib/python3.7/dist-packages (from multipledispatch>=0.4.7->datashape>=0.5.1->datashader) (1.15.0)
Requirement already satisfied: llvmlite<0.35,>=0.34.0.dev0 in /usr/local/lib/python3.7/dist-packages (from numba>=0.51->datashader) (0.34.0)
Requirement already satisfied: pyparsing!=3.0.5,>=2.0.2 in /usr/local/lib/python3.7/dist-packages (from packaging>=16.8->bokeh) (3.0.9)
Requirement already satisfied: pytz>=2017.3 in /usr/local/lib/python3.7/dist-packages (from pandas>=0.24.1->datashader) (2022.1)
Requirement already satisfied: locket in /usr/local/lib/python3.7/dist-packages (from partd>=0.3.10->dask[complete]>=0.18.0->datashader) (1.0.0)
Requirement already satisfied: importlib-metadata in /usr/local/lib/python3.7/dist-packages (from xarray>=0.9.6->datashader) (4.11.4)
Requirement already satisfied: heapdict in /usr/local/lib/python3.7/dist-packages (from zict>=0.1.3->distributed>=2.0->dask[complete]>=0.18.0->datashader) (1.0.1)
Requirement already satisfied: pyviz-comms>=0.7.4 in /usr/local/lib/python3.7/dist-packages (from holoviews) (2.2.0)
Requirement already satisfied: panel>=0.13.1 in /usr/local/lib/python3.7/dist-packages (from holoviews) (0.13.1)
Requirement already satisfied: requests in /usr/local/lib/python3.7/dist-packages (from panel>=0.13.1->holoviews) (2.23.0)
Requirement already satisfied: bleach in /usr/local/lib/python3.7/dist-packages (from panel>=0.13.1->holoviews) (5.0.0)
Requirement already satisfied: markdown in /usr/local/lib/python3.7/dist-packages (from panel>=0.13.1->holoviews) (3.3.7)
Requirement already satisfied: tqdm>=4.48.0 in /usr/local/lib/python3.7/dist-packages (from panel>=0.13.1->holoviews) (4.64.0)
Requirement already satisfied: webencodings in /usr/local/lib/python3.7/dist-packages (from bleach->panel>=0.13.1->holoviews) (0.5.1)
Requirement already satisfied: zipp>=0.5 in /usr/local/lib/python3.7/dist-packages (from importlib-metadata->xarray>=0.9.6->datashader) (3.8.0)
Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.7/dist-packages (from requests->panel>=0.13.1->holoviews) (2022.6.15)
Requirement already satisfied: chardet<4,>=3.0.2 in /usr/local/lib/python3.7/dist-packages (from requests->panel>=0.13.1->holoviews) (3.0.4)
Requirement already satisfied: urllib3!=1.25.0,!=1.25.1,<1.26,>=1.21.1 in /usr/local/lib/python3.7/dist-packages (from requests->panel>=0.13.1->holoviews) (1.24.3)
Requirement already satisfied: idna<3,>=2.5 in /usr/local/lib/python3.7/dist-packages (from requests->panel>=0.13.1->holoviews) (2.10)
In [32]:
%matplotlib inline

import pandas as pd
import datashader as ds
from datashader import transfer_functions as tf
from colorcet import fire

Because the dataset is pretty big, let's use a small sample first. For this visualization, we only keep the dropoff location.

Usually, we use remotezip package in Python to download and extract the big dataset. But one of the problem with remotezip is that it does not support range request and that is why we have to download the dataset manually. We suggest you to download the zip file of dataset containing csv from Kaggle dataset, extract it and put the filepath of CSV file in the csv_path variable below.

In [33]:
from google.colab import drive
drive.mount('/content/drive')  
Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount("/content/drive", force_remount=True).
In [34]:
dirpath = '/content/drive/MyDrive/Data/data_Viz/'
# dirpath = '/content/drive/MyDrive/Data/'
!ls $dirpath
nyc_taxi_data_2014.csv
In [35]:
csv_path= dirpath + 'nyc_taxi_data_2014.csv'

try:
    nyctaxi_small = pd.read_csv(csv_path, nrows=10000, 
                                usecols=['dropoff_longitude', 'dropoff_latitude'])
except:
    print("Dataset URL is not correct or not defined:")
    print("Creating dummy dataset so that code won't break but for assignment, you must use actual dataset.")
    # nyctaxi_small = pd.DataFrame({"dropoff_longitude": [-73, -74], "dropoff_latitude": [40, 41]})
    
nyctaxi_small.head()
Out[35]:
dropoff_longitude dropoff_latitude
0 -73.982227 40.731790
1 -73.960449 40.763995
2 -73.986626 40.765217
3 -73.979863 40.777050
4 -73.984367 40.720524

Although the dataset is different, we can still follow the example here: https://datashader.org/getting_started/Introduction.html

In [36]:
agg = ds.Canvas().points(nyctaxi_small, 'dropoff_longitude', 'dropoff_latitude')
tf.set_background(tf.shade(agg, cmap=fire),"black")
Out[36]:

Why can't we see anything? Wait, do you see the small dots on the left top? Can that be New York City? Maybe we don't see anything because some people travel very far? or because the dataset has some missing data?

Q: Can you first check whether there are NaNs? Then drop them and draw the map again?

In [37]:
# Implement: Check whether we have NaNs
nyctaxi_small.isna().sum()
Out[37]:
dropoff_longitude    1
dropoff_latitude     1
dtype: int64
In [38]:
# Implement: drop the rows with NaN and then draw the map again. 
nyctaxi_small = nyctaxi_small.dropna(axis=0)
nyctaxi_small.isna().sum()
Out[38]:
dropoff_longitude    0
dropoff_latitude     0
dtype: int64
In [39]:
agg = ds.Canvas().points(nyctaxi_small, 'dropoff_longitude', 'dropoff_latitude')
tf.set_background(tf.shade(agg, cmap=fire),"black")
Out[39]:

So it's not about the missing data.

Q: Can you identify the issue and draw the map like the following?

hint: https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.Series.between.html and histograms may be helpful.

In [40]:
# nyctaxi_small.describe()
In [41]:
# nyctaxi_small.info()
In [42]:
import matplotlib.pyplot as plt

# nyctaxi_small.plot.hist()

plt.hist(nyctaxi_small.dropoff_longitude)
plt.hist(nyctaxi_small.dropoff_latitude)
Out[42]:
(array([ 182.,    0.,    0.,    0.,    0.,    0.,    0.,    0.,    0.,
        9817.]),
 array([ 0.       ,  4.1071847,  8.2143694, 12.3215541, 16.4287388,
        20.5359235, 24.6431082, 28.7502929, 32.8574776, 36.9646623,
        41.071847 ]),
 <a list of 10 Patch objects>)
In [43]:
nyctaxi_small[nyctaxi_small['dropoff_longitude']==0].shape
Out[43]:
(182, 2)

New York/Coordinates 40.7128° N, 74.0060° W

https://www.kaggle.com/code/muhammedazamkhan/interactive-data-visualization-nyc-taxi-trip/notebook

(-74.05, -73.7), (40.6, 40.9)

In [44]:
# nyctaxi_small_filtered = nyctaxi_small[nyctaxi_small['dropoff_longitude'].between(-74.05, -73.7)  ]

# nyctaxi_small_filtered = nyctaxi_small_filtered[nyctaxi_small_filtered['dropoff_latitude'].between(40.6, 40.9)]
In [45]:
# Implement. You can use multiple cells to figure out what's going on. 
# TODO: Once you figure it out, Replace below dummy value of df nyctaxi_small_filtered with correct value where the issue is resolved

nyctaxi_small_filtered = nyctaxi_small[nyctaxi_small['dropoff_longitude'].between(-74.1, -73.7)  ]
nyctaxi_small_filtered = nyctaxi_small_filtered[nyctaxi_small_filtered['dropoff_latitude'].between(40, 40.9)]
In [46]:
agg = ds.Canvas().points(nyctaxi_small_filtered, 'dropoff_longitude', 'dropoff_latitude')
tf.set_background(tf.shade(agg, cmap=fire), "black")
Out[46]:
In [47]:
agg = ds.Canvas().points(nyctaxi_small_filtered, 'dropoff_longitude', 'dropoff_latitude')
tf.set_background(tf.shade(agg, cmap=fire), "black")
Out[47]:

Do you see the black empty space at the center? That looks like the Central Park. This is cool, but it'll be awesome if we can explore the data interactively.

Q. Ok, now let's get serious by loading the whole dataset. It may take some time. Apply the same data cleaning procedure.

In [48]:
%%time
# Implement 
nyctaxi_small = pd.read_csv(csv_path, usecols=['dropoff_longitude', 'dropoff_latitude'])
CPU times: user 26.6 s, sys: 1.71 s, total: 28.3 s
Wall time: 42.4 s

Can you feed the data directly to datashader to reproduce the static plot, this time with the full data?

In [49]:
# %%time
nyctaxi_small_filtered = nyctaxi_small[nyctaxi_small['dropoff_longitude'].between(-74.1, -73.7)  ]
nyctaxi_small_filtered = nyctaxi_small_filtered[nyctaxi_small_filtered['dropoff_latitude'].between(40.5, 41)]

agg = ds.Canvas().points(nyctaxi_small_filtered, 'dropoff_longitude', 'dropoff_latitude')
tf.set_background(tf.shade(agg, cmap=fire), "black")
Out[49]:

Wow, that's fast. Also it looks cool!

Let's try the interactive version from here: https://datashader.org/getting_started/Introduction.html

In [50]:
nyctaxi_small.shape    , nyctaxi_small_filtered.shape
Out[50]:
((14999999, 2), (14751421, 2))
In [51]:
import holoviews as hv
from holoviews.element.tiles import EsriImagery
from holoviews.operation.datashader import datashade
hv.extension('bokeh')

map_tiles  = EsriImagery().opts(alpha=0.5, width=900, height=480, bgcolor='black')
points     = hv.Points(nyctaxi_small_filtered, ['dropoff_longitude', 'dropoff_latitude'])
taxi_trips = datashade(points, x_sampling=1, y_sampling=1, cmap=fire, width=900, height=480)

map_tiles * taxi_trips
Out[51]:

Why does it say "map data not yet available"? The reason is the difference between two coordinate systems. If you google this error message, you can find https://stackoverflow.com/questions/44487898/map-background-with-datashader-map-data-not-yet-available.

You can use datashader.utils.lnglat_to_meters to convert your latitudes and longitudes to a format that holoviews understands. More on this here: https://datashader.org/user_guide/Geography.html

Q: Can you draw an interactive map by converting the lnglat data to x, y coordinate explained above?

In [52]:
nyctaxi_small_filtered.head()
Out[52]:
dropoff_longitude dropoff_latitude
0 -73.982227 40.731790
1 -73.960449 40.763995
2 -73.986626 40.765217
3 -73.979863 40.777050
4 -73.984367 40.720524
In [53]:
%%time
nyctaxi_small_mercator = pd.DataFrame(ds.utils.lnglat_to_meters(nyctaxi_small_filtered['dropoff_longitude'], nyctaxi_small_filtered['dropoff_latitude']))
CPU times: user 2min 59s, sys: 3.69 s, total: 3min 3s
Wall time: 3min 14s
In [54]:
nyctaxi_small_mercator = nyctaxi_small_mercator.T
nyctaxi_small_mercator.head()
Out[54]:
dropoff_longitude dropoff_latitude
0 -8.235664e+06 4.972861e+06
1 -8.233240e+06 4.977593e+06
2 -8.236154e+06 4.977773e+06
3 -8.235401e+06 4.979512e+06
4 -8.235902e+06 4.971206e+06
In [55]:
# https://stackoverflow.com/questions/44487898/map-background-with-datashader-map-data-not-yet-available
# https://github.com/holoviz/datashader/blob/master/datashader/utils.py
# datashader.utils.lnglat_to_meters()

# import holoviews as hv
# from holoviews.element.tiles import EsriImagery
# from holoviews.operation.datashader import datashade
hv.extension('bokeh')

map_tiles  = EsriImagery().opts(alpha=0.5, width=900, height=480, bgcolor='black')
points     = hv.Points(nyctaxi_small_mercator, ['dropoff_longitude', 'dropoff_latitude'])
taxi_trips = datashade(points, x_sampling=1, y_sampling=1, cmap=fire, width=900, height=480)

map_tiles * taxi_trips
Out[55]:
In [55]:
 
In [56]:
# Implement

It's interactive! Actually, if you are running a bokeh server and there is a live python process, the map quickly refreshes and show more details as you zoom.

Q: how many rows (data points) are we visualizing right now?

In [57]:
# figure it out
nyctaxi_small_mercator.shape
Out[57]:
(14751421, 2)

We are visualizing 14751421 data points

That's a lot of data points. If we are using a vector format, it is probably hopeless to expect any interactivity because you need to move that many points! Yet, datashader + holoviews + bokeh renders everything almost in real time!

Leaflet

Another useful tool is Leaflet. It allows you to use various map tile data (Google maps, Open streetmap, ...) with many types of marks (points, heatmap, etc.). Leaflet.js is one of the easiest options to do that on the web, and there is a Python bridge of it: https://github.com/jupyter-widgets/ipyleaflet. Although we will not go into details, it's certainly something that's worth checking out if you're using geographical data.

In [57]:
 

Convert the notebook to HTML

In [58]:
# %%shell
# jupyter nbconvert --to html /content/m12_lab_Ganapathy_Anitha.ipynb